thank you! this fixed the issue I was having

I'd also like this merged.

Thank you! That solved my problem.

@city96 Is there anything else you need done here?

I don't pretend to understand all of what is going on here, but I've found that this only causes an error if these two layers are in BF16; if they are F32 or F16 they seem to work fine. Not sure why that is. Or, I may be experiencing an unrelated issue.

this should be on the main.

While this fixes loading for zimage turbo, I am getting similar errors for zimage base.
@vaclavmuller, can you have a look?
These are the affected models:
https://huggingface.co/babakarto/z-image-base-gguf/resolve/main/z_image_base_BF16.gguf
https://huggingface.co/babakarto/z-image-base-gguf/resolve/main/z_image_base_Q8_0.gguf

z_image_base_Q8_0.gguf:
The expanded size of the tensor (3840) must match the existing size (4080) at non-singleton dimension 1. Target sizes: [3840, 3840]. Tensor sizes: [3840, 4080]

z_image_base_BF16.gguf
The expanded size of the tensor (3840) must match the existing size (7680) at non-singleton dimension 1. Target sizes: [3840, 3840]. Tensor sizes: [3840, 7680]